package com.tianji.learning.excel.easyexcel.conveter;

import com.alibaba.excel.converters.Converter;
import com.alibaba.excel.converters.ReadConverterContext;
import com.alibaba.excel.converters.WriteConverterContext;
import com.alibaba.excel.enums.CellDataTypeEnum;
import com.alibaba.excel.metadata.data.WriteCellData;

/**
 * @ClassName AnonymityConverter
 * @Description 为 EasyExcel工具提供的数据转换器，性别转换器   true-是，false=否
 * @Author admin
 * @Version 2.0
 **/
public class AnonymityConverter implements Converter<Boolean> {

    /**
     * 实体类中对象属性类型
     */
    @Override
    public Class<?> supportJavaTypeKey() {
        return Boolean.class;
    }

    /**
     * Excel中对应的CellData属性类型
     */
    @Override
    public CellDataTypeEnum supportExcelTypeKey() {
        return CellDataTypeEnum.STRING;
    }

    /**
     * 这里读的时候会调用 -> 导入
     */
    public Boolean convertToJavaData(ReadConverterContext<?> context) {
        // 从Cell中读取数据
        String collegeStr = context.getReadCellData().getStringValue();
        // 判断Excel中的值，将其转换为预期的数值
        if ("是".equals(collegeStr)) {
            return true;
        } else if ("否".equals(collegeStr)) {
            return false;
        }
        return false;
    }

    /**
     * 这里写的时候会调用 -> 导出
     */
    public WriteCellData<?> convertToExcelData(WriteConverterContext<Boolean> context) {
        Boolean anonymity = context.getValue();
        if (anonymity) {
            return new WriteCellData<>("是");
        } else {
            return new WriteCellData<>("否");
        }
    }
}























